package com.linyaonan.leetcode.easy._766;

/**
 * 如果一个矩阵的每一方向由左上到右下的对角线上具有相同元素，那么这个矩阵是托普利茨矩阵。
 * <p>
 * 给定一个 M x N 的矩阵，当且仅当它是托普利茨矩阵时返回 True。
 * <p>
 * 示例 1:
 * <p>
 * 输入:
 * matrix = [
 *   [1,2,3,4],
 *   [5,1,2,3],
 *   [9,5,1,2]
 * ]
 * 输出: True
 * 解释:
 * 在上述矩阵中, 其对角线为:
 * "[9]", "[5, 5]", "[1, 1, 1]", "[2, 2, 2]", "[3, 3]", "[4]"。
 * 各条对角线上的所有元素均相同, 因此答案是True。
 * 示例 2:
 * <p>
 * 输入:
 * matrix = [
 *   [1,2],
 *   [2,2]
 * ]
 * 输出: False
 * 解释:
 * 对角线"[1, 2]"上的元素不同。
 * 说明:
 * <p>
 *  matrix 是一个包含整数的二维数组。
 * matrix 的行数和列数均在 [1, 20]范围内。
 * matrix[i][j] 包含的整数在 [0, 99]范围内。
 *
 * @author: Lin
 * @date: 2020/1/5
 */
public class ToeplitzMatrix {
    int[][] m;
    int maxX;
    int maxY;
    public boolean isToeplitzMatrix(int[][] matrix) {
        m = matrix;
        maxX = matrix.length - 1;
        maxY = matrix[0].length - 1;
        for (int i = 0; i <= maxY; i++) {
            if (!topToEndIsToeplitz(0, i)) {
                return false;
            }
        }
        for (int i = 1; i <= maxX; i++) {
            if (!topToEndIsToeplitz(i, 0)) {
                return false;
            }
        }
        return true;
    }

    private boolean topToEndIsToeplitz(int x, int y) {
        int start = m[x][y];
        while (x >= 0 && x <= maxX && y >= 0 && y <= maxY) {
            if (start != m[x][y]) {
                return false;
            } else {
                x++;
                y++;
            }
        }
        return true;
    }
}
